export dataset_name=/home/user4/workspace/ddg-benchmark/data

# progen2_base progen2_BFD90 progen2_large progen2_medium progen2_oas progen2_small progen2_xlarge
CUDA_VISIBLE_DEVICES=6 python src/Progen2/predict.py \
    --dataset_input $dataset_name \
    --model_dir src/Progen2/models \
    --model_locate progen2_base progen2_BFD90 progen2_large progen2_medium progen2_oas progen2_small progen2_xlarge